package atom.core0.util.security;

import java.security.SecureRandom;
import java.util.Scanner;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

import atom.core0.util.SunBase64;

public class JAES {
	private static final String CIPHER_ALGORITHM_ECB = "AES/ECB/PKCS5Padding";
	private static final String CIPHER_ALGORITHM_CBC = "AES/CBC/ZerosPadding";
	private String m_Key;
	private String m_IV;//IV length: must be 16 bytes long
	private SecretKey key;

	public void setEcbKey(String key) throws Exception {
		this.m_Key = key;
		convertKey(key);
	}
	public void setCbcKey(String key) throws Exception {
		this.m_Key = key;
		convertKey(key);
	}

	//是根据随机源 构造密钥  相同的随机源则相同的密钥
	public void convertKey(String keys) throws Exception {
		// 1.构造密钥生成器，指定为AES算法,不区分大小写
		KeyGenerator keygen = KeyGenerator.getInstance("AES");
		// 2.根据ecnodeRules规则初始化密钥生成器
		// 生成一个128位的随机源,根据传入的字节数组
		keygen.init(128, new SecureRandom(m_Key.getBytes()));
		// 3.产生原始对称密钥
		SecretKey original_key = keygen.generateKey();
		// 4.获得原始对称密钥的字节数组
		byte[] raw = original_key.getEncoded();
		// 5.根据字节数组生成AES密钥
		key = new SecretKeySpec(raw, "AES");

	}

	//key 必须是16字节
	public void convertKey2(String keys) throws Exception {
		key = new SecretKeySpec(keys.getBytes(), "AES");
	}

	/*
	 * 加密 1.构造密钥生成器 2.根据ecnodeRules规则初始化密钥生成器 3.产生密钥 4.创建和初始化密码器 5.内容加密 6.返回字符串
	 */
	public byte[] ecbEncode(byte[] byte_encode) throws Exception {
		
		// 6.根据指定算法AES自成密码器
		Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM_ECB);
		// 7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密解密(Decrypt_mode)操作，第二个参数为使用的KEY
		cipher.init(Cipher.ENCRYPT_MODE, key);
		// 9.根据密码器的初始化方式--加密：将数据加密
		byte[] byte_AES = cipher.doFinal(byte_encode);
		return byte_AES;

	}

	/*
	 * 解密 解密过程： 1.同加密1-4步 2.将加密后的字符串反纺成byte[]数组 3.将加密内容解密
	 */
	public byte[] ecbDecode(byte[] byte_content) throws Exception {
		// 6.根据指定算法AES自成密码器
		Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM_ECB);
		// 7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密(Decrypt_mode)操作，第二个参数为使用的KEY
		cipher.init(Cipher.DECRYPT_MODE, key);
		/*
		 * 解密
		 */
		byte[] byte_decode = cipher.doFinal(byte_content);
		return byte_decode;

	}
	public byte[] cbcEncode(byte[] byte_encode) throws Exception {
		
		// 6.根据指定算法AES自成密码器
		Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM_CBC);
		// 7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密解密(Decrypt_mode)操作，第二个参数为使用的KEY
		cipher.init(Cipher.ENCRYPT_MODE, key,new IvParameterSpec(m_IV.getBytes()));
		// 9.根据密码器的初始化方式--加密：将数据加密
		byte[] byte_AES = cipher.doFinal(byte_encode);
		return byte_AES;

	}
	public byte[] cbcDecode(byte[] byte_content) throws Exception {
		// 6.根据指定算法AES自成密码器
		Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM_CBC);
		// 7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密(Decrypt_mode)操作，第二个参数为使用的KEY
		cipher.init(Cipher.DECRYPT_MODE, key,new IvParameterSpec(m_IV.getBytes()));
		/*
		 * 解密
		 */
		byte[] byte_decode = cipher.doFinal(byte_content);
		return byte_decode;

	}

}
